../../notebooks/solutions/_derived/4_get_logits_stroke.py